<!DOCTYPE html>
<html lang="en">

<head>
    <meta charset="UTF-8">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <title>4.5.1-嵌套的effect与effect栈</title>
</head>

<body>
    <div id="container"></div>
</body>
<script>
    /**
     * effect 是可以发生嵌套的
     * 
     * 问题: 
     * - 用全局变量 activeEffect 来存储通过 effect 函数注册的副作用函数，
     *    这意味着同一时刻 activeEffect 所存储的副作用函数只能有一个
     * - 当副作用函数发生嵌套时，内层副作用函数的执行会覆盖 activeEffect 的值，
     *    并且永远不会恢复到原来的值
     * - 这时如果再有响应式数据进行依赖收集，即使这个响应式数据是在外层副作用函数中读取的，
     *    它们收集到的副作用函数也都会是内层副作用函数
     * 
     * 解决：
     * - 需要一个副作用函数栈 effectStack，
     *    在副作用函数执行时，将当前副作用函数压入栈中
     * - 待副作用函数执行完毕后将其从栈中弹出，
     *    并始终让 activeEffect 指向栈顶的副作用函数
     * - 当前执行的副作用函数会被压入栈顶，这样当副作用函数发生嵌套时，
     *    栈底存储的就是外层副作用函数，而栈顶存储的则是内层副作用函数
     * 
     * */

    // 存储副作用函数的桶
    const bucket = new WeakMap()
    // 原始数据
    const data = { foo: true, bar: true } // 新增 修改
    // 对原始数据的代理
    const obj = new Proxy(data, {
        // 拦截读取操作
        get(target, key) {
            // 将副作用函数 activeEffect 添加到存储副作用函数的桶中
            track(target, key)
            // 返回属性值
            return target[key]
        },
        // 拦截设置操作
        set(target, key, newVal) {
            // 设置属性值
            target[key] = newVal
            // 把副作用函数从桶里取出并执行
            trigger(target, key)
        }
    })

    // 在 get 拦截函数内调用 track 函数追踪变化
    function track(target, key) {
        // 没有 activeEffect，直接 return
        if (!activeEffect) return
        let depsMap = bucket.get(target)
        if (!depsMap) {
            bucket.set(target, (depsMap = new Map()))
        }
        let deps = depsMap.get(key)
        if (!deps) {
            depsMap.set(key, (deps = new Set()))
        }
        // 把当前激活的副作用函数添加到依赖集合 deps 中
        deps.add(activeEffect)
        // deps 就是一个与当前副作用函数存在联系的依赖集合
        // 将其添加到 activeEffect.deps 数组中
        activeEffect.deps.push(deps)
    }
    // 在 set 拦截函数内调用 trigger 函数触发变化
    function trigger(target, key) {
        const depsMap = bucket.get(target)
        if (!depsMap) return
        const effects = depsMap.get(key)

        const effectsToRun = new Set(effects)
        effectsToRun.forEach(effectFn => { effectFn() })
        // effects && effects.forEach(fn => fn()) // 问题出在这句代码
    }

    // 用一个全局变量存储被注册的副作用函数
    let activeEffect
    // effect 栈
    // 个人理解，始终都会留着一个是最先执行的
    const effectStack = []  // 新增

    function effect(fn) {
        const effectFn = () => {
            // 调用 cleanup 函数完成清除工作
            cleanup(effectFn)
            // 当调用 effect 注册副作用函数时，将副作用函数赋值给 activeEffect
            activeEffect = effectFn
            // 在调用副作用函数之前将当前副作用函数压入栈中
            effectStack.push(effectFn)  // 新增
            fn()
            // 在当前副作用函数执行完毕后，将当前副作用函数弹出栈，并把 activeEffect 还原为之前的值
            effectStack.pop()  // 新增
            activeEffect = effectStack[effectStack.length - 1]  // 新增
        }
        // activeEffect.deps 用来存储所有与该副作用函数相关的依赖集合
        effectFn.deps = []
        // 执行副作用函数
        effectFn()
    }

    function cleanup(effectFn) {
        // 遍历 effectFn.deps 数组
        for (let i = 0; i < effectFn.deps.length; i++) {
            // deps 是依赖集合
            const deps = effectFn.deps[i]
            // 将 effectFn 从依赖集合中移除
            deps.delete(effectFn)
        }
        // 最后需要重置 effectFn.deps 数组
        effectFn.deps.length = 0
    }

    // 全局变量
    let temp1, temp2
    
    // effectFn1 嵌套了 effectFn2
    effect(function effectFn1() {
        console.log('effectFn1 执行')
        effect(function effectFn2() {
            console.log('effectFn2 执行')
            // 在 effectFn2 中读取 obj.bar 属性
            temp2 = obj.bar
        })
        // 在 effectFn1 中读取 obj.foo 属性
        temp1 = obj.foo
    })

    setTimeout(() => {
        // 修改 obj.bar 属性
        obj.bar = false
    }, 1000)
</script>

</html>